change param of fetch_origin from name to id

Brightcells 8 年之前
父節點
當前提交
ee2f21f685
共有 1 個文件被更改,包括 15 次插入3 次删除
  1. 15 3
      main.py

+ 15 - 3
main.py

@@ -114,6 +114,18 @@ def get_new_files(lensman, session, maxt):
114 114
     return news
115 115
 
116 116
 
117
+def get_origin_path_from_id(lensman, session, id_):
118
+    origin, _ = get_session_dir(lensman, session)
119
+    files = glob.iglob('{}/*'.format(origin))
120
+    for file in files:
121
+        if not file.endswith('.tmp'):  # Whether 'xxx.tmp' or not
122
+            filename = file.split('/')[-1]
123
+            name = filename.split('.')[0]
124
+            if name == id_:
125
+                return file.strip(ROOT_PATH)
126
+    return ''
127
+
128
+
117 129
 class HelloHandler(RequestHandler):
118 130
     def get(self):
119 131
         self.write('Hello Tornado')
@@ -171,14 +183,14 @@ class FetchOriginHandler(RequestHandler):
171 183
         lensman = self.get_argument('lensman', '')
172 184
         session = self.get_argument('session', '')
173 185
 
174
-        name = self.get_argument('name', '')
186
+        id_ = self.get_argument('id', '')
175 187
 
176
-        origin, _ = get_session_dir(lensman, session)
188
+        path = get_origin_path_from_id(lensman, session, id_)
177 189
 
178 190
         self.write({
179 191
             'status': 200,
180 192
             'data': {
181
-                'path': '{}/{}'.format(origin, name).strip(ROOT_PATH),
193
+                'path': path,
182 194
             }
183 195
         })
184 196